package com.inspur.edp.lcm.metadata.core.thread;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.inspur.edp.lcm.metadata.api.exception.ErrorCodes;
import com.inspur.edp.lcm.metadata.api.exception.LcmConfigResolveDevException;
import com.inspur.edp.lcm.metadata.common.FileServiceImp;
import com.inspur.edp.lcm.metadata.common.Utils;
import org.springframework.util.StringUtils;

import java.io.IOException;
import java.util.HashSet;
import java.util.Set;

/*
 * Copyright (c) 2020 - present, Inspur Genersoft Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
public class BoGraphThreadLocalHolder {
    public static ThreadLocal<Set<String>> dependencyGraphBoPathSet = InheritableThreadLocal.withInitial(() -> null);

    private static final String FILE_NAME = "config/platform/dev/main/ide_bographcache.json";
    private static final String SECTION_NAME = "speedSwitch";
    private static ObjectMapper objectMapper = new ObjectMapper();
    private static  String absoluteFilePath;

    private static String getAbsoluteFilePath() {
        if (StringUtils.isEmpty(absoluteFilePath)) {
            absoluteFilePath = Utils.getAbsoluteConfigFilePath(FILE_NAME);
        }
        return absoluteFilePath;
    }

    /**
     * 是否开启工程画图缓存提速开关
     *
     * @return
     */
    private static boolean isSpeedSwitchEnabled() {
        try {

            FileServiceImp fileService = new FileServiceImp();
            String fileContents = fileService.fileRead(getAbsoluteFilePath());
            JsonNode jsonNode = objectMapper.readTree(fileContents);
            return jsonNode.findValue(SECTION_NAME).asBoolean(true);
        } catch (IOException e) {
            throw new LcmConfigResolveDevException(e, ErrorCodes.ECP_CONFIG_RESOLVE_DEV_0002, getAbsoluteFilePath());
        }
    }

    /**
     * 初始化
     */
    public static void init() {
        if(isSpeedSwitchEnabled()){
            dependencyGraphBoPathSet.set(new HashSet<>());
        }
    }

    /**
     * 清除
     */
    public static void clear() {
        dependencyGraphBoPathSet.remove();
    }

    public static boolean contains(String boPath) {
        if (dependencyGraphBoPathSet.get() == null) {
            return false;
        }
        return dependencyGraphBoPathSet.get().contains(boPath);
    }

    public static void put(String boPath) {
        if (dependencyGraphBoPathSet.get() == null) {
            return;
        }
        dependencyGraphBoPathSet.get().add(boPath);
    }

    public static void remove(String boPath) {
        if (dependencyGraphBoPathSet.get() == null) {
            return;
        }
        dependencyGraphBoPathSet.get().remove(boPath);
    }
}
